from tensorflow.keras.models import load_model

from bert4keras.snippets import AutoRegressiveDecoder

seq2seq_model = load_model('./best_model')
seq2seq_model.summary()